package com.mdnote.practice.dp.stock;

/**
 * @author Rhythm-2019
 * @version 1.0
 * @date 2020/9/22
 * @description 买卖股票的最佳时机 II
 */
public class LeetCode122 {

    public static void main(String[] args) {
        LeetCode122 leetCode122 = new LeetCode122();
        int res = leetCode122.maxProfit(new int[]{7, 1, 5, 3, 6, 4});
        System.out.println(res);
    }

    public int maxProfit(int[] prices) {

        if (prices.length <= 1) {
            return 0;
        }
        int[][] dp = new int[prices.length][2];

        dp[0][0] = 0;
        dp[0][1] = -prices[0];

        for (int i = 1; i < dp.length; i++) {
            // 这天没有股票，昨天卖了或者昨天就没有持有
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            // 当天手上有股票，昨天买的或者昨天就有了（这里就可以买，第一题k等于1不能卖）
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }

        return dp[dp.length - 1][0];
    }
}
